package com.lw.common.codegen;

import com.alibaba.druid.pool.DruidDataSource;
import com.mybatisflex.codegen.Generator;
import com.mybatisflex.codegen.config.GlobalConfig;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * mybatis-flex-codegen 代码自动生成工具
 */
public class FlexCodegen {
    public static void main(String[] args) {
        //配置数据源
        DruidDataSource dataSource= new DruidDataSource();
        dataSource.setUrl("jdbc:mysql://localhost:3306/springboot3mybatisflex?useUnicode=true&characterEncoding=utf-8&serverTimezone=UTC&useSSL=false&nullCatalogMeansCurrent=true&allowPublicKeyRetrieval=true");
        dataSource.setUsername("root");
        dataSource.setPassword("123456");

        //创建配置内容，两种风格都可以。
        // 默认配置
        GlobalConfig globalConfig = createGlobalConfigUseStyle1();

        //通过 datasource 和 globalConfig 创建代码生成器
        Generator generator = new Generator(dataSource, globalConfig);

        //生成代码
        generator.generate();
    }
    public static GlobalConfig createGlobalConfigUseStyle1() {
        //创建配置内容
        GlobalConfig globalConfig = new GlobalConfig();

        //设置根包
        globalConfig.setBasePackage("com.lw");

        //设置表前缀和只生成哪些表
        // 填写需要生成数据库和表
        globalConfig.setGenerateTable("tb_big_data");

        //设置生成 entity
        globalConfig.setEntityGenerateEnable(true);

        //带swagger
        globalConfig.setEntityWithSwagger(true);

        globalConfig.setEntityOverwriteEnable(true);

        // 启用 Lombok
        globalConfig.setEntityWithLombok(false);
        //设置生成 mapper
        globalConfig.setMapperGenerateEnable(true);
        // 控制类
        globalConfig.setControllerGenerateEnable(true);
        globalConfig.setControllerRestStyle(true);
        // 服务类
        globalConfig.setServiceGenerateEnable(true);
        // 服务实现类
        globalConfig.setServiceImplGenerateEnable(true);
        // 文件
        globalConfig.setMapperXmlGenerateEnable(false);
        // 表定义辅助类 ,涉及到多表查询的时候回用到的
        globalConfig.setTableDefGenerateEnable(true);

        // 注释配置
        Date date = new Date();
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy年MM月dd日 HH:mm:ss");
        String format = dateFormat.format(date);
        globalConfig.getJavadocConfig()
                .setAuthor("lw")
                .setSince(format);
        return globalConfig;
    }
}
